package com.kevinkk.tree;

/**
 * 给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
 *
 * 采用后序遍历，实现类似从底向上的遍历效果
 */

public class LowestCommonAncestorOfABinaryTree {
    class Solution {
        public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
            if (root == null) return null;
            // 找到了 p 或者 q 节点
            if (root == p || root == q) return root;

            TreeNode left = lowestCommonAncestor(root.left, p, q);
            TreeNode right = lowestCommonAncestor(root.right, p, q);
            if (left == null && right == null) return null; // 都没找到
            if (left != null && right != null) return root; // 都找到了
            return left == null ? right : left;             // 找到一个
        }
    }
}
